/*!
 * \file __frproto_add_list.c
 *
 * All Rights Reserved Copyright (C) 2014 broada,Co.,Ltd.\n
 *
 * \version 3.0
 * \author geyh
 * \date 2016/03/31
 *
 */
#include <CUnit/Basic.h>
#include <stdlib.h>

#include "frcomm.h"
#include "frmem.h"
#include "frlist.h"
#include "frproto_mng.h"
#include "apmtest_cunit.h"

#include "apmtest_comm.h"

static int __suite_init(void)
{
//    frmodule_mng_init();
    return 0;
}

static int __suite_cleanup(void)
{
//    frmodule_mng_destory();
    return 0;
}

extern int g_i_flag_run;

typedef struct __st_proto_mng {
    int i_is_enable;            /* 指定协议时有效 FR_YES,FR_NO */
    unsigned int ui_ver;        /* 指定协议时有效,协议版本号(更新后 +1) */
    int i_ip_proto;             /* 6(IPPROTO_TCP) / 17(IPPROTO_UDP) */
    u_int ui_ip;                /* ip 地址 */
    u_short us_port_start;      /* port 开始 */
    u_short us_port_end;        /* port 结束 */

    Frlist_header st_hdr_module_mng; /* module链表 */
} Frproto_mng;

void __hash_node_mng_free(Frlist_node *p_st_node);

int __frproto_add_list(Frhash_header *g_p_st_hash
        , int i_ip_proto
        , u_int ui_ip
        , u_short us_port_start
        , u_short us_port_end
        , int i_l7_proto
        , int i_is_enable);

static void __case1(void)
{
    Frmodule_mng *p_st_module_mng=NULL;
    Frlist_node *p_st_module_mng_node = NULL;
    extern Frlist_header *g_p_st_hdr_mng_module_use;
    Frhash_header *g_p_st_hash = NULL;
    int i_ip_proto = 6;
    u_int ui_ip = iptools_parse_ipaddr((char *)"192.168.1.1");
    u_short us_port_start = 40;
    u_short us_port_end = 80;
    int i_l7_proto = 1;
    int i_ret = 0;

    g_i_flag_run = 1;
    frmodule_mng_init();
    p_st_module_mng = (Frmodule_mng *)frmalloc(sizeof(Frmodule_mng));
    memset(p_st_module_mng, 0x00, sizeof(Frmodule_mng));
    p_st_module_mng->p_st_l7_module = (L7_module *)frmalloc(sizeof(L7_module));
    memset(p_st_module_mng->p_st_l7_module, 0x00, sizeof(L7_module));
    p_st_module_mng->p_st_l7_module->i_l7_proto = 1;

    p_st_module_mng_node = frlist_node_create(p_st_module_mng);
    frlist_add_tail(g_p_st_hdr_mng_module_use,p_st_module_mng_node);
    g_p_st_hash = frhash2_create2(5);

    frproto_mng_init();
    i_ret = __frproto_add_list(g_p_st_hash, i_ip_proto, ui_ip
            , us_port_start, us_port_end, i_l7_proto, FR_YES);

    CU_ASSERT_EQUAL(i_ret,0);

    g_i_flag_run = 0;
    frmodule_mng_destory();
    frhash2_free2(g_p_st_hash, __hash_node_mng_free);
    frproto_mng_destory();

}

static void __case2(void)
{
    Frmodule_mng *p_st_module_mng=NULL;
    Frlist_node *p_st_module_mng_node = NULL;
    extern Frlist_header *g_p_st_hdr_mng_module_use;
    Frhash_header *g_p_st_hash = NULL;
    int i_ip_proto = 6;
    u_int ui_ip = iptools_parse_ipaddr((char *)"192.168.1.1");
    u_short us_port_start = 60;
    u_short us_port_end = 100;
    int i_l7_proto = 1;
    int i_ret = 0;

    g_i_flag_run = 1;
    frmodule_mng_init();
    p_st_module_mng = (Frmodule_mng *)frmalloc(sizeof(Frmodule_mng));
    memset(p_st_module_mng, 0x00, sizeof(Frmodule_mng));
    p_st_module_mng->p_st_l7_module = (L7_module *)frmalloc(sizeof(L7_module));
    memset(p_st_module_mng->p_st_l7_module, 0x00, sizeof(L7_module));
    p_st_module_mng->p_st_l7_module->i_l7_proto = 1;

    p_st_module_mng_node = frlist_node_create(p_st_module_mng);
    frlist_add_tail(g_p_st_hdr_mng_module_use,p_st_module_mng_node);
    g_p_st_hash = frhash2_create2(5);

    frproto_mng_init();
    i_ret = __frproto_add_list(g_p_st_hash, i_ip_proto, ui_ip
            , us_port_start, us_port_end, i_l7_proto, FR_YES);


    us_port_start = 40;
    us_port_end = 80;
    i_ret = __frproto_add_list(g_p_st_hash, i_ip_proto, ui_ip
            , us_port_start, us_port_end, i_l7_proto, FR_YES);

    CU_ASSERT_EQUAL(i_ret,3);

    g_i_flag_run = 0;
    frmodule_mng_destory();
    frhash2_free2(g_p_st_hash, __hash_node_mng_free);
    frproto_mng_destory();

}


static void __case3(void)
{
    Frmodule_mng *p_st_module_mng=NULL;
    Frlist_node *p_st_module_mng_node = NULL;
    extern Frlist_header *g_p_st_hdr_mng_module_use;
    Frhash_header *g_p_st_hash = NULL;
    int i_ip_proto = 6;
    u_int ui_ip = iptools_parse_ipaddr((char *)"192.168.1.1");
    u_short us_port_start = 40;
    u_short us_port_end = 80;
    int i_l7_proto = 1;
    int i_ret = 0;

    g_i_flag_run = 1;
    frmodule_mng_init();
    p_st_module_mng = (Frmodule_mng *)frmalloc(sizeof(Frmodule_mng));
    memset(p_st_module_mng, 0x00, sizeof(Frmodule_mng));
    p_st_module_mng->p_st_l7_module = (L7_module *)frmalloc(sizeof(L7_module));
    memset(p_st_module_mng->p_st_l7_module, 0x00, sizeof(L7_module));
    p_st_module_mng->p_st_l7_module->i_l7_proto = 1;

    p_st_module_mng_node = frlist_node_create(p_st_module_mng);
    frlist_add_tail(g_p_st_hdr_mng_module_use,p_st_module_mng_node);
    g_p_st_hash = frhash2_create2(5);

    frproto_mng_init();
    i_ret = __frproto_add_list(g_p_st_hash, i_ip_proto, ui_ip
            , us_port_start, us_port_end, i_l7_proto, FR_YES);

    i_ret = __frproto_add_list(g_p_st_hash, i_ip_proto, ui_ip
            , us_port_start, us_port_end, i_l7_proto, FR_YES);

    CU_ASSERT_EQUAL(i_ret,0);

    g_i_flag_run = 0;
    frmodule_mng_destory();
    frhash2_free2(g_p_st_hash, __hash_node_mng_free);
    frproto_mng_destory();

}

static void __case4(void)
{
    Frhash_header *g_p_st_hash = NULL;
    int i_ip_proto = 6;
    u_int ui_ip = iptools_parse_ipaddr((char *)"192.168.1.1");
    u_short us_port_start = 40;
    u_short us_port_end = 80;
    int i_l7_proto = 1;
    int i_ret = 0;

    g_i_flag_run = 1;
    frmodule_mng_init();

    g_p_st_hash = frhash2_create2(5);

    frproto_mng_init();
    i_ret = __frproto_add_list(g_p_st_hash, i_ip_proto, ui_ip
            , us_port_start, us_port_end, i_l7_proto, FR_YES);

    i_ret = __frproto_add_list(g_p_st_hash, i_ip_proto, ui_ip
            , us_port_start, us_port_end, i_l7_proto, FR_YES);

    CU_ASSERT_EQUAL(i_ret,2);

    g_i_flag_run = 0;
    frmodule_mng_destory();
    frhash2_free2(g_p_st_hash, __hash_node_mng_free);
    frproto_mng_destory();

}

static void __case5(void)
{
    Frmodule_mng *p_st_module_mng=NULL;
    Frlist_node *p_st_module_mng_node = NULL;
    extern Frlist_header *g_p_st_hdr_mng_module_use;
    Frhash_header *g_p_st_hash = NULL;
    int i_ip_proto = 6;
    u_int ui_ip = iptools_parse_ipaddr((char *)"192.168.1.1");
    u_short us_port_start = 40;
    u_short us_port_end = 80;
    int i_l7_proto = 1;
    int i_ret = 0;

    g_i_flag_run = 1;
    frmodule_mng_init();
    p_st_module_mng = (Frmodule_mng *)frmalloc(sizeof(Frmodule_mng));
    memset(p_st_module_mng, 0x00, sizeof(Frmodule_mng));
    p_st_module_mng->p_st_l7_module = (L7_module *)frmalloc(sizeof(L7_module));
    memset(p_st_module_mng->p_st_l7_module, 0x00, sizeof(L7_module));
    p_st_module_mng->p_st_l7_module->i_l7_proto = 1;

    p_st_module_mng_node = frlist_node_create(p_st_module_mng);
    frlist_add_tail(g_p_st_hdr_mng_module_use,p_st_module_mng_node);

    frproto_mng_init();
    i_ret = __frproto_add_list(g_p_st_hash, i_ip_proto, ui_ip
            , us_port_start, us_port_end, i_l7_proto, FR_YES);

    i_ret = __frproto_add_list(g_p_st_hash, i_ip_proto, ui_ip
            , us_port_start, us_port_end, i_l7_proto, FR_YES);

    CU_ASSERT_EQUAL(i_ret,1);

    g_i_flag_run = 0;
    frmodule_mng_destory();
    frhash2_free2(g_p_st_hash, __hash_node_mng_free);
    frproto_mng_destory();

}

int add_suite___frproto_add_list(void)
{
    CU_pSuite pSuite = NULL;

    if ((pSuite = CU_add_suite("frproto_mng __frproto_add_list()", __suite_init, __suite_cleanup)) == NULL)
        return FR_FAIL;

    CU_add_test(pSuite, "case1", __case1);
    CU_add_test(pSuite, "case2", __case2);
    CU_add_test(pSuite, "case3", __case3);
    CU_add_test(pSuite, "case4", __case4);
    CU_add_test(pSuite, "case5", __case5);

    return FR_SUCC;
}
